from django.contrib import admin
from django.db.models import Q

from ffpms_app.models import Task, Area

# Register your models here.

admin.site.site_header = admin.site.site_header + ' 农户建档绩效管理'


# 任务Admin类
@admin.register(Task)
class TaskAdmin(admin.ModelAdmin):
    # 列表显示字段
    list_display = ['id', 'title', 'start_time',
                    'end_time', 'num', 'manager', 'executor']

    def get_queryset(self, request):
        # tasks = Task.objects.all() # 全量任务数据
        queryset = super().get_queryset(request)
        user = request.user  # 当前用户
        if not user.is_superuser:  # 非 超级管理员
            # 显示 任务管理者（客户经理） 或 任务执行者（业务员）相关的任务
            # queryset = Task.objects.filter(Q(manager=user) | Q(executor=user))
            queryset = queryset.filter(Q(manager=user) | Q(executor=user))
        return queryset


# 区域Admin类
@admin.register(Area)
class AreaAdmin(admin.ModelAdmin):
    # 列表显示字段
    list_display = ['id', 'name', 'created_time', 'last_modified']
